from plat.plat_train import get_train_info, plat_train_val, plat_trains

"""
比较是否使用loop_type特征和是否使用密码子的效果
codon_0: 未使用密码子; codon_1: 使用了密码子
loop_0: 未使用loop_type特征; loop_1: 使用了loop_type特征
pos_enc: 在使用loop_type特征和密码子的情况下加入位置编码的效果
"""
if __name__ == '__main__':
    log_path = "../log/GatedGCN/loop/"
    codon_0_loop_0 = "codon_0_loop_0.log"
    codon_0_loop_1 = "codon_0_loop_1.log"
    codon_1_loop_0 = "codon_1_loop_0.log"
    codon_1_loop_1 = "codon_1_loop_1.log"
    pos_enc = "pos_enc.log"
    train_dict = dict()
    val_dict = dict()
    train_dict['codon_0_loop_0'], val_dict['codon_0_loop_0'] = get_train_info(log_path + codon_0_loop_0)
    train_dict['codon_0_loop_1'], val_dict['codon_0_loop_1'] = get_train_info(log_path + codon_0_loop_1)
    train_dict['codon_1_loop_0'], val_dict['codon_1_loop_0'] = get_train_info(log_path + codon_1_loop_0)
    train_dict['codon_1_loop_1'], val_dict['codon_1_loop_1'] = get_train_info(log_path + codon_1_loop_1)
    train_dict['pos_enc'], val_dict['pos_enc'] = get_train_info(log_path + pos_enc)
    plat_trains(train_dict)
    plat_trains(val_dict)
